package com.hfb.mashibing.alip8.diamondsquarealgorithm.class04;

import java.util.Arrays;

/**
 * 算法：
 *      kmp：KMP 算法（Knuth-Morris-Pratt 算法）
 *      是一个著名的字符串匹配算法，效率很高，但是确实有点复杂。
 *  个人见解：
 *      KMP 算法永不回退 txt 的指针 i，不走回头路（不会重复扫描 txt），
 *      而是借助 dp 数组中储存的信息把 pat 移到正确的位置继续匹配，时间复杂度只需 O(N)，
 *      用空间换时间，所以我认为它是一种动态规划算法。
 *
 */
public class Code01_KMP {

    /**
     * @param str 文本串
     * @param dest 模式串
     * @param next 匹配核心数组
     * @return
     */
    public static int kmp(String str, String dest,int[] next) {
        for(int i = 0, j = 0; i < str.length(); i++){
            if (j > 0 && str.charAt(i) != dest.charAt(j)) {
                j = next[j - 1];
            }
            if (str.charAt(i) == dest.charAt(j)) {
                j++;
            }
            if (j == dest.length()) {
                return i-j+1;
            }
        }
        return 0;
    }

    public static int[] kmpnext(String dest) {
        int[] next = new int[dest.length()];
        next[0] = 0;
        for(int i = 1,j = 0; i < dest.length(); i++) {
            if (j > 0 && dest.charAt(j) != dest.charAt(i)) {
                j = next[j - 1];
            }
            if (dest.charAt(i) == dest.charAt(j)) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }

    public static void main(String[] args){
        String a = "ABABAE";
        String b = "ABABABABAEBEABADAEABAEABABAE";
        int[] next = kmpnext(a);
        System.out.println(Arrays.toString(next));
        int res = kmp(b, a,next);
        System.out.println(res);
    }
}
